fix auto restart issue